package cn.jinzhu.netty.client;

import cn.jinzhu.netty.message.*;
import cn.jinzhu.netty.protocol.MessageCodec;
import cn.jinzhu.netty.protocol.MessageCodecSharable;
import cn.jinzhu.netty.protocol.ProcotolFrameDecoder;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;
import io.netty.handler.timeout.IdleState;
import io.netty.handler.timeout.IdleStateEvent;
import io.netty.handler.timeout.IdleStateHandler;
import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Scanner;
import java.util.Set;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;

@Slf4j
public class ChatClient {
    public static void main(String[] args) {
        NioEventLoopGroup group = new NioEventLoopGroup();
        //@Sharable 可共享的handler 表示在netty中这个类可以共享给所有的NioEventLoop,说明表示可以安全 自定义协议
        MessageCodecSharable MESSAGE_CODEC = new MessageCodecSharable();
        //自带日志
        LoggingHandler LOGGINGH_ANDLER = new LoggingHandler(LogLevel.DEBUG);

        //线程通信  计数+1 表示阻塞 -1不阻塞 这个类使一个线程等待其他线程各自执行完毕后再执行
        //好比 线程1 是0  就阻塞了 不会向下执行了 线程2拿到了 给latch-1 那么就会唤醒线程1  因为线程1变成了0
        CountDownLatch latch = new CountDownLatch(1);

        //可以使线程原子性 在多线程中只能有一个线程拿到该属性 好比一个线程拿到了 改成了fasle
        // 其他线程在获取的时候就是false 不同同时一起操作  （好比线程1 拿到了false  在改成true 那么线程b就只能获取到true ,类似加了锁）
        AtomicBoolean atomicBoolean = new AtomicBoolean(false);
        try {
            Bootstrap bootstrap = new Bootstrap();
            bootstrap.channel(NioSocketChannel.class);
            bootstrap.group(group);
            bootstrap.handler(new ChannelInitializer<SocketChannel>() {
                @Override
                protected void initChannel(SocketChannel ch) throws Exception {

                    // 解决服务端发送过来的数据半包粘包问题  入栈处理器
                    ch.pipeline().addLast(new ProcotolFrameDecoder());
                    //用来判断是不是 读空闲时间过长，或 写空闲时间过长
                    // 3s 内如果没有向服务器写数据，会触发一个 IdleState#WRITER_IDLE 事件
                    ch.pipeline().addLast(new IdleStateHandler(0, 3, 0));

                    // ChannelDuplexHandler 可以同时作为入站和出站处理器
                    ch.pipeline().addLast(new ChannelDuplexHandler() {
                        // 用来触发特殊事件
                        @Override
                        public void userEventTriggered(ChannelHandlerContext ctx, Object evt) throws Exception{
                            IdleStateEvent event = (IdleStateEvent) evt;
                            // 触发了写空闲事件
                            if (event.state() == IdleState.WRITER_IDLE) {
                                //心跳包 表示证明我虽然服务端你检测到我没有发送数据但是我要证明我还活着
                               ctx.writeAndFlush(new PingMessage());
                            }
                        }
                    });
                    //打印服务端数据 出栈处理器
                    ch.pipeline().addLast(LOGGINGH_ANDLER);
                    //将数据放入自定义协议正文中 出栈处理器
                    ch.pipeline().addLast(MESSAGE_CODEC);
                    //业务handler处理器  入栈处理器调用后会触发出栈处理器  出栈是从tail尾部向上找出栈处理器 入栈则是重头开始向下找
                    ch.pipeline().addLast("client handler ", new ChannelInboundHandlerAdapter() {
                        //接收服务端数据
                        @Override
                        public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                            System.out.println("asdadasdd" + msg.toString());
                            if ((msg instanceof LoginResponseMessage)) {
                                //获取登陆消息
                                LoginResponseMessage message = (LoginResponseMessage) msg;
                                if (message.isSuccess()) {
                                    //将该线程加入
                                    atomicBoolean.set(true);
                                }
                            }
                            //线程通信计数-1  （接收到了登陆返回的消息）  唤醒用户输入线程await
                            latch.countDown();
                        }

                        //登陆  调用建立连接之后立马执行的handler处理器  (连接建立后触发事件)
                        @Override
                        public void channelActive(ChannelHandlerContext ctx) throws Exception {
                            //用户输入要新建线程 毕竟会阻塞 要与netty线程不相干
                            new Thread(() -> {
                                Scanner scanner = new Scanner(System.in);
                                System.out.println("请输入用户名");
                                String username = scanner.nextLine();
                                System.out.println("请输入密码");
                                String passowrd = scanner.nextLine();
                                //创建消息对象
                                LoginRequestMessage loginessage = new LoginRequestMessage(username, passowrd);
                                //发送给服务端
                                ctx.writeAndFlush(loginessage);
                                System.out.println("操作");
                                try {
                                    //如果是1表示阻塞 0表示启动不会阻塞
                                    latch.await();
                                } catch (InterruptedException e) {
                                    e.printStackTrace();
                                }
                                //登陆失败
                                if (!atomicBoolean.get()) {
                                    System.out.println("进来");
                                    //关闭channel
                                    ctx.channel().close();
                                    return;
                                }
                                //登陆成功
                                while (true) {
                                    System.out.println("==================================");
                                    System.out.println("send [username] [content]");
                                    System.out.println("gsend [group name] [content]");
                                    System.out.println("gcreate [group name] [m1,m2,m3...]");
                                    System.out.println("gmembers [group name]");
                                    System.out.println("gjoin [group name]");
                                    System.out.println("gquit [group name]");
                                    System.out.println("quit");
                                    System.out.println("==================================");
                                    String command = null;
                                    try {
                                        command = scanner.nextLine();
                                    } catch (Exception e) {
                                        break;
                                    }
//                                    if (atomicBoolean.get()) {
//                                        return;
//                                    }
                                    String[] s = command.split(" ");
                                    switch (s[0]) {
                                        case "send":
                                            System.out.println("send发送消息");
                                            //聊天请求  发给谁
                                            ctx.writeAndFlush(new ChatRequestMessage(username, s[1], s[2]));
                                            break;
                                        case "gsend":
                                            //发送群聊请求
                                            ctx.writeAndFlush(new GroupChatRequestMessage(username, s[1], s[2]));
                                            break;
                                        case "gcreate":
                                            //创建聊天室请求
                                            Set<String> set = new HashSet<>(Arrays.asList(s[2].split(",")));
                                            set.add(username); // 加入自己
                                            ctx.writeAndFlush(new GroupCreateRequestMessage(s[1], set));
                                            break;
                                        case "gmembers":
                                            //小组成员请求
                                            ctx.writeAndFlush(new GroupMembersRequestMessage(s[1]));
                                            break;
                                        case "gjoin":
                                            //加入聊天室请求
                                            ctx.writeAndFlush(new GroupJoinRequestMessage(username, s[1]));
                                            break;
                                        case "gquit":
                                            //退出聊天室
                                            ctx.writeAndFlush(new GroupQuitRequestMessage(username, s[1]));
                                            break;
                                        case "quit":
                                            //退出
                                            ctx.channel().close();
                                            return;
                                    }

                                }

                            }, "system in").start();
                        }
                        // 在连接断开时触发
                        @Override
                        public void channelInactive(ChannelHandlerContext ctx) throws Exception {
                            log.debug("连接已经断开，按任意键退出..");
                            //设置这个是为了打开锁 避免死锁  因为其他线程要拿到这个锁
                            atomicBoolean.set(true);
                        }

                        // 在出现异常时触发
                        @Override
                        public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
                            log.debug("连接已经断开，按任意键退出..{}", cause.getMessage());
                            //设置这个是为了打开锁 避免死锁  因为其他线程要拿到这个锁
                            atomicBoolean.set(true);
                        }

                    });

                }
            });
            Channel channel = bootstrap.connect("localhost", 8080).sync().channel();
            //获取关闭事件  同步获取
            channel.closeFuture().sync();
        } catch (Exception e) {
            log.error("client error", e);
        } finally {
            group.shutdownGracefully();
        }
    }



}
